# +----------------------------------------------------------------------
# | DjangoAdmin敏捷开发框架 [ 赋能开发者，助力企业发展 ]
# +----------------------------------------------------------------------
# | 版权所有 2021~2024 北京DjangoAdmin研发中心
# +----------------------------------------------------------------------
# | Licensed LGPL-3.0 DjangoAdmin并不是自由软件，未经许可禁止去掉相关版权
# +----------------------------------------------------------------------
# | 官方网站: https://www.djangoadmin.cn
# +----------------------------------------------------------------------
# | 作者: @一米阳光 团队荣誉出品
# +----------------------------------------------------------------------
# | 版权和免责声明:
# | 本团队对该软件框架产品拥有知识产权（包括但不限于商标权、专利权、著作权、商业秘密等）
# | 均受到相关法律法规的保护，任何个人、组织和单位不得在未经本团队书面授权的情况下对所授权
# | 软件框架产品本身申请相关的知识产权，禁止用于任何违法、侵害他人合法权益等恶意的行为，禁
# | 止用于任何违反我国法律法规的一切项目研发，任何个人、组织和单位用于项目研发而产生的任何
# | 意外、疏忽、合约毁坏、诽谤、版权或知识产权侵犯及其造成的损失 (包括但不限于直接、间接、
# | 附带或衍生的损失等)，本团队不承担任何法律责任，本软件框架禁止任何单位和个人、组织用于
# | 任何违法、侵害他人合法利益等恶意的行为，如有发现违规、违法的犯罪行为，本团队将无条件配
# | 合公安机关调查取证同时保留一切以法律手段起诉的权利，本软件框架只能用于公司和个人内部的
# | 法律所允许的合法合规的软件产品研发，详细声明内容请阅读《框架免责声明》附件；
# +----------------------------------------------------------------------

from fastapi import APIRouter
from fastapi.logger import logger
from sqlalchemy import and_
from starlette.requests import Request
from starlette.responses import HTMLResponse, RedirectResponse

from apps.models.user import User
from apps.services import menu, user
from apps.views import render
from config.env import FASTAPI_DEMO
from extends import db
from utils import R
from utils.utils import getImageURL, uid

# 创建路由
router = APIRouter()


# 渲染主页
@router.get('/', summary='主页', response_class=HTMLResponse)
@router.get('/index', summary='主页', response_class=HTMLResponse)
async def index(request: Request):
    try:
        # 获取用户信息
        user_id = uid(request)
        # 查询用户信息
        userInfo = db.query(User).filter(and_(User.id == user_id, User.is_delete == 0)).first()
        if userInfo:
            # 头像
            userInfo.avatar = getImageURL(userInfo.avatar)
        # 获取菜单列表
        menuList = menu.GetPermissionMenuList(user_id)
        # 模板参数
        content = {
            'request': request,
            'userInfo': userInfo,
            'menuList': menuList,
        }
        # 使用render渲染方式
        return render('index.html', context=content)
    except:
        # 抛出异常
        logger.error('运行异常')
        raise
    finally:
        # 关闭连接
        db.close()


# 欢迎页
@router.get('/main', summary='欢迎页', response_class=HTMLResponse)
async def main(request: Request):
    # 使用render渲染方式
    return render('main.html', context={'request': request})


# 个人信息
@router.api_route(methods=['GET', 'POST'], path="/userInfo", summary='个人信息')
async def userInfo(request: Request):
    if request.method == "GET":
        try:
            # 获取当前用户信息
            user_info = db.query(User).filter(and_(User.id == uid(request), User.is_delete == 0)).first()
            if user_info:
                # 用户头像
                user_info.avatar = getImageURL(user_info.avatar)
            # 模板参数
            content = {
                'request': request,
                'user': user_info,
            }
            # 使用render渲染方式
            return render('user_info/index.html', context=content)
        except:
            # 抛出异常
            logger.error('运行异常')
            raise
        finally:
            # 关闭连接
            db.close()
    elif request.method == "POST":
        if FASTAPI_DEMO:
            return R.failed("演示环境，暂无操作权限")
        # 调用更新用户服务方法
        result = await user.UserInfo(request)
        # 返回结果
        return result


# 修改密码
@router.post('/updatePwd', summary='修改密码')
async def updatePwd(request: Request):
    if FASTAPI_DEMO:
        return R.failed("演示环境，暂无操作权限")
    # 调用更新用户服务方法
    result = await user.UpdatePwd(request)
    # 返回结果
    return result


# 退出登录
@router.get('/logout', summary='退出登录')
async def logout(request: Request):
    # 删除登录ID
    request.session.clear()
    request.cookies.clear()
    # 重定向到登录页
    return RedirectResponse("/index")
